import { contextBridge, ipcRenderer } from 'electron'
import { electronAPI } from '@electron-toolkit/preload'

// Custom APIs for renderer
const app = {
  sendMessage: (channel, data) => {
    ipcRenderer.send(channel, data)
  },
  onMessage: (channel, callback) => {
    const listener = (event, ...args) => callback(...args)
    ipcRenderer.on(channel, listener)
    // 返回一个函数，用于移除监听器
    return () => {
      ipcRenderer.off(channel, listener)
    }
  },
  // 切换主题
  useDark: (isDark) => {
    ipcRenderer.send('useDark', isDark)
  },
  // 退出应用
  quit: () => {
    ipcRenderer.invoke('quit')
  },
  // 重启应用
  restart: () => {
    ipcRenderer.invoke('restart')
  },
  // 获取系统信息
  getSystemInfo: () => {
    return ipcRenderer.invoke('getSystemInfo')
  },
  // 设置配置信息
  setConfig(config, type) {
    const cType = type || 'app'
    return ipcRenderer.invoke('setConfig', config, cType)
  },
  // 获取配置信息
  getConfig: (type) => {
    const cType = type || 'app'
    return ipcRenderer.invoke('getConfig', cType)
  }
}

const bot = {
  friendshipAdd: (phone) => {
    return ipcRenderer.invoke('friendship:add', phone)
  }
}

// Use `contextBridge` APIs to expose Electron APIs to
// renderer only if context isolation is enabled, otherwise
// just add to the DOM global.
if (process.contextIsolated) {
  try {
    contextBridge.exposeInMainWorld('electron', electronAPI)
    contextBridge.exposeInMainWorld('app', app)
    contextBridge.exposeInMainWorld('bot', bot)

    // 在渲染进程中暴露安全的 IPC 方法
  } catch (error) {
    console.error(error)
  }
} else {
  window.electron = electronAPI
  window.app = app
  window.bot = bot
}
